package com.project.controller;

import com.project.bean.UserBean;
import com.project.service.IUserService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.servlet.http.HttpServletRequest;

@RestController
@RequestMapping("user")
public class UserController {

    @Autowired
    private IUserService userService;
    @RequestMapping("login")
    public String login(String userName, String pwd, HttpServletRequest request){
        UserBean user = userService.login(userName, pwd);
        if (user == null) {
            return "no";
        }else {
            request.getSession().setAttribute("loginUser",user);
            return "ok";
        }
    }

    @RequestMapping("getLoginUser")
    public UserBean getLoginUser(HttpServletRequest request){
        return (UserBean) request.getSession().getAttribute("loginUser");
    }

    @RequestMapping("register")
    public String register(UserBean user){
        boolean register = userService.register(user);
        if (register == true) {
            return "ok";
        }else {
            return "no";
        }
    }


}
